package com.zhiyou.app.service;

import cn.hutool.json.JSONObject;
import com.zhiyou.app.pojo.entity.TaskEntity;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

import java.io.IOException;
import java.util.List;
import java.util.Map;

public interface ComfyUIService {
    String uploadFile(String fileUrl);

    void doTask(String promptJson, TaskEntity task, String clientId) throws IOException;

    String view(JSONObject res, TaskEntity taskEntity);

    Map<String, List<String>> getRes(String promptId, TaskEntity taskEntity);

    void getTaskProcessTracker(TaskEntity taskEntity, String clientId, String promptId, String orderNum);
}
